package client

import (
	"fmt"
	"os"
)

var userId int
var userPwd string

func PrintMenu() {
	// 接收用户的选择
	var key int
	// 判断是否还继续显示菜单
	var loop = true
	for {
		if !loop {
			break
		}
		fmt.Println("-------------------------欢迎登陆多人聊天系统------------------------")
		fmt.Println("\t\t\t\t 1 登录聊天室")
		fmt.Println("\t\t\t\t 2 注册用户")
		fmt.Println("\t\t\t\t 3 退出系统")
		fmt.Print("\t\t\t\t 请选择(1-3): ")

		n, err := fmt.Scanf("%d\n", &key)
		if n != 1 || err != nil {
			fmt.Println("输入错误")
		}

		switch key {
		case 1:
			fmt.Println("登录聊天室...")
			loop = false
		case 2:
			fmt.Println("注册用户...")
			loop = false
		case 3:
			fmt.Println("退出系统...")
			os.Exit(0)
		default:
			fmt.Println("您的输入有误，请重新输入...")
		}
	}

	// 根据用户的输入，显示新的提示信息
	if key == 1 {
		// 说明用户要登录
		fmt.Print("请输入用户id: ")
		n, err := fmt.Scanf("%d\n", &userId)
		if n != 1 || err != nil {
			fmt.Println("输入错误")
		}
		fmt.Print("请输入用户的密码: ")
		n, err = fmt.Scanf("%s\n", &userPwd)
		if n != 1 || err != nil {
			fmt.Println("输入错误")
		}

		// 把登录函数，写入到另外一个文件
		err = login(userId, userPwd)
		if err != nil {
			fmt.Println(err, "登录失败")
		} else {
			fmt.Println("登录成功")
		}
	} else {

	}
}
